import tensorflow as tf
A = tf.reshape(tf.range(20, dtype=tf.float32), (5, 4))
print(A)
print(tf.reduce_sum(A))